/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.streamsets.datacollector.client.cli.command.store;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.streamsets.datacollector.client.ApiClient;
import com.streamsets.datacollector.client.api.StoreApi;
import com.streamsets.datacollector.client.cli.command.BaseCommand;
import com.streamsets.datacollector.client.cli.command.PipelineConfigAndRulesJson;
import com.streamsets.datacollector.client.model.PipelineConfigurationJson;
import com.streamsets.datacollector.client.model.RuleDefinitionsJson;
import io.airlift.airline.Command;
import io.airlift.airline.Option;

import java.io.File;

@Command(name = "export", description = "Export Pipeline Configuration & Rules")
public class ExportPipelineCommand extends BaseCommand {
  @Option(
    name = {"-n", "--name"},
    description = "Pipeline Name",
    required = true
  )
  public String pipelineName;

  @Option(
    name = {"-r", "--revision"},
    description = "Pipeline Revision",
    required = false
  )
  public String pipelineRev;

  @Option(
    name = {"-f", "--file"},
    description = "Export file name",
    required = true
  )
  public String fileName;

  public void run() {
    if(pipelineRev == null) {
      pipelineRev = "0";
    }

    ApiClient apiClient = getApiClient();
    StoreApi storeApi = new StoreApi(apiClient);
    try {
      ObjectMapper mapper = apiClient.getJson().getMapper();

      PipelineConfigurationJson pipelineConfigurationJson = storeApi.getPipelineInfo(pipelineName, pipelineRev,
        "pipeline", false);

      RuleDefinitionsJson ruleDefinitionsJson = storeApi.getPipelineRules(pipelineName, pipelineRev);

      PipelineConfigAndRulesJson pipelineConfigAndRulesJson = new PipelineConfigAndRulesJson(pipelineConfigurationJson,
        ruleDefinitionsJson);

      mapper.writeValue(new File(fileName), pipelineConfigAndRulesJson);

      System.out.println("Successfully exported pipeline '" + pipelineName + "' to file - " + fileName );
    } catch(Exception e) {
      e.printStackTrace();
    }
  }
}
